"""用来存储公共的功能"""""
from functools import wraps

from flask import current_app
from flask import g
from flask import session



"""自定义过滤器,来完成热门新闻颜色效果"""""
def do_index_class(index):
	if index == 1:
		return "first"
	elif index == 2:
		return "second"
	elif index == 3:
		return "third"
	else:
		return ""




"""登陆状态装饰器"""
def user_login_data(view_func):
	@wraps(view_func)
	def wrapper(*args,**kwargs):
		# 获取用户编号
		user_id = session.get("user_id")
		# 通过用户编号查询用户对象
		user = None
		if user_id:
			try:
				from info.models import User
				user = User.query.get(user_id)
			except Exception as e:
				current_app.logger.error(e)

		"""使用全局变量g保存user"""
		g.user = user
		return view_func(*args,**kwargs)
	return wrapper
